package leetcode;

/*
766. 托普利茨矩阵
如果矩阵上每一条由左上到右下的对角线上的元素都相同，那么这个矩阵是 托普利茨矩阵 。
给定一个 M x N 的矩阵，当且仅当它是托普利茨矩阵时返回 True。
示例 1:
输入:
matrix = [
  [1,2,3,4],
  [5,1,2,3],
  [9,5,1,2]
]
输出: True
解释:
在上述矩阵中, 其对角线为:
"[9]", "[5, 5]", "[1, 1, 1]", "[2, 2, 2]", "[3, 3]", "[4]"。
各条对角线上的所有元素均相同, 因此答案是True。
示例 2:
输入:
matrix = [
  [1,2],
  [2,2]
]
输出: False
解释:
对角线"[1, 2]"上的元素不同。
说明:
matrix 是一个包含整数的二维数组。
matrix 的行数和列数均在 [1, 20]范围内。
matrix[i][j] 包含的整数在 [0, 99]范围内
*/

import java.util.HashMap;
import java.util.Map;

public class problems_766 {
    public static void main(String[] args) {
        System.out.println(new Solution().isToeplitzMatrix(new int[][]{
                {1, 2, 3, 4},
                {5, 1, 2, 3},
                {9, 5, 1, 2}
        }));
    }

    static class Solution {
        public boolean isToeplitzMatrix(int[][] matrix) {
            // key是对角线的key值，val是任意值
            Map<Integer, Integer> mapValue = new HashMap<>();
            for (int i = 0; i < matrix.length; i++) {
                for (int j = 0; j < matrix[0].length; j++) {
                    if (!mapValue.containsKey(i - j)) {
                        mapValue.put(i - j, matrix[i][j]);
                    } else if (mapValue.get(i - j) != matrix[i][j]) {
                        return false;
                    }
                }
            }
            return true;
        }
    }
}